// 设置画布

const canvas = document.querySelector('canvas');
const ctx = canvas.getContext('2d');

/*
浏览器窗口的视口（viewport）高/宽度（以像素为单位）；
如果有滚动条，也包括滚动条高/宽度。
*/
const width = canvas.width = window.innerWidth;
const height = canvas.height = window.innerHeight;

// 生成随机数的函数
function random(min, max) {
  const num = Math.floor(Math.random() * (max - min)) + min;
  return num;
}

// 生成随机的颜色
function randomColor() {
  return 'rgb(' + random(0, 255) + ',' +
    random(0, 255) + ',' +
    random(0, 255) + ')'
}

// 生成小球
/**
 * @description: 
 * @param {*} x 小球的横坐标
 * @param {*} y 小球的纵坐标
 * @param {*} velX 水平速度
 * @param {*} velY 垂直速度
 * @param {*} color 小球颜色
 * @param {*} size 小球的半径
 * @return {*}
 */
function Ball(x, y, velX, velY, color, size) {
  this.x = x;
  this.y = y;
  this.velX = velX;
  this.velY = velY;
  this.color = color;
  this.size = size;
}

// 画出小球的方法
Ball.prototype.draw = function () {
  // 声明我们现在要开始在纸上画一个图形了
  ctx.beginPath();
  ctx.fillStyle = this.color;
  ctx.arc(this.x, this.y, this.size, 0, 2 * Math.PI)
  ctx.fill()
}

// 更新小球的位置 设置边缘
Ball.prototype.update = function () {
  if ((this.x - this.size) <= 0 || (this.x + this.size) >= width) {
    this.velX = -this.velX
  }
  if ((this.y - this.size) <= 0 || (this.y + this.size) >= height) {
    this.velY = -this.velY
  }
  this.x += this.velX
  this.y += this.velY
}

// 添加碰撞检测
Ball.prototype.collisionDetect = function () {
  // 遍历 balls[] 数组中的小球。
  for (let j = 0; j < balls.length; j++) {
    // if语句判断调用方法的小球是否为当前遍历的小球 
    if (this !== balls[j]) {
      // distance为两个小球中心的距离
      const dx = this.x - balls[j].x;
      const dy = this.y - balls[j].y;
      // 开平方根
      const distance = Math.sqrt(dx * dx + dy * dy);

      // 两个小球中心的距离小于两小球的半径和则说明两小球相撞
      if (distance < this.size + balls[j].size) {
        // 将两个小球的颜色都设置成随机的一种
        balls[j].color = this.color = randomColor();
      }
    }
  }
}
// 存储小球
let balls = [];
while (balls.length < 25) {
  let size = random(10, 20)
  let ball = new Ball(
    // 球至少离画布边缘球本身一倍宽度的距离
    random(0 + size, width - size),
    random(0 + size, height - size),
    random(-7, 7),
    random(-7, 7),
    randomColor(),
    size
  )
  balls.push(ball)
}

// 运动循环 每一帧都自动更新视图
function loop() {
  // 填充的颜色设置成半透明的rgba(0,0,0,0.25)
  // 也就是让之前的视图留下来一点点
  // 从而可以看到小球运动时的轨迹
  ctx.fillStyle = 'rgba(0, 0, 0, 0.25)';
  ctx.fillRect(0, 0, width, height);

  for (let i = 0; i < balls.length; i++) {
    balls[i].draw();
    balls[i].update();
    // 检测小球碰撞
    balls[i].collisionDetect();
  }
  // requestAnimationFrame() 方法再运行一次函数 递归
  requestAnimationFrame(loop);
}
loop();